Rec-Distill: An Industrial Distillation Pipeline for Large-Scale Recommendation Models
Rec-Distill:面向工业推荐的大规模模型蒸馏框架
🧭 推荐系统 · 知识蒸馏 · 大规模模型压缩 ★ 深度解读
ByteDance AML · ByteDance
解耦式蒸馏框架将大模型(24B参数、20K序列)知识高效迁移至轻量线上模型,迁移率超60%并取得商业收益
📌 工作参考价值(4 点)
- 解耦式蒸馏推进大模型复用:教师与学生对训练完全解耦,教师持续流式更新并缓存logits,学生异步消费。可参考此“1-to-N”范式,一个大教师同时服务多个小模型(如召回、粗排、精排),显著降低训练成本,实现线上模型不限制实验规模。
- 黑盒蒸馏+loss缩放:使用教师输出logits的交叉熵损失,简单有效,且观察到蒸馏损失尺度比主任务损失小两个数量级时效果差,需通过调大系数α对齐量级。实际部署时可直接采用CE蒸馏,并手动调大α。
- 去偏机制解决样本分布不匹配:教师和学生训练数据采样策略不同导致logits偏差,通过将教师logits用学生侧去偏公式重校正(𝑇2′=𝑓𝑆(𝑇1)),再与学生去偏logits蒸馏,可提点0.1%。电商场景普遍存在采样偏差,该方法可复用。
- 混合批-流式管道保证实时性:批训练阶段用大量数据快速收敛,流式阶段从高速存储读取教师最新logits追加到学生流数据,实现分钟级延迟适应数据分布变化,避免线上掉点。适合实时推荐系统的高频更新需求。